List comprehension using index values